package com.spzx.common.rabbit.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

/**
 * 服务器启动时，执行rabbitTemplate初始化，设置确认函数和回退函数
 * ApplicationEvent      一些子事件的父类。
 * ApplicationReadyEvent 具体子事件。表示应用程序启动好，IOC容器初始化好，存在相关bean对象了。再进行相关的初始化。
 * 也可以使用相关的注解替代： @EventListener
 */
@Slf4j
@Component
public class RabbitInitConfigApplicationListener implements ApplicationListener<ApplicationReadyEvent> {

    @Autowired
    RabbitTemplate rabbitTemplate;

    @Autowired
    RedisTemplate redisTemplate;


    @Override
    public void onApplicationEvent(ApplicationReadyEvent event) {
        this.setUpInitRabbitTemplete();
    }


    private void setUpInitRabbitTemplete() {
        //注册确认回调函数
        rabbitTemplate.setConfirmCallback((new RabbitTemplate.ConfirmCallback() {
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                //发消息时如果携带了关联数据，会把关联数据退回来；
                //如果发消息没有携带关联数据,那么，就是null
                log.info("setConfirmCallback - correlationData=" + correlationData);
                if(ack){
                    //交换机收到消息
                    log.info("交换机收到消息");
                }else{
                    log.info("交换机未收到消息");
                    log.error("原因:"+cause);
                }
            }
        }));
        //注册回退回调函数
        rabbitTemplate.setReturnsCallback(new RabbitTemplate.ReturnsCallback() {
            @Override
            public void returnedMessage(ReturnedMessage returned) {
                log.info("returned"+returned);
                //消息重发，如何获取关联数据？ 关联数据没有回退。可以从缓存中获取。
                String correlationId = returned.getMessage()
                        .getMessageProperties().getHeader("spring_returned_message_correlation");
                String correlationDataStr = (String) redisTemplate.opsForValue().get(correlationId);

            }
        });
    }


}
